numpy hackerrank solution